//输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构) 
//
// B是A的子结构， 即 A中有出现和B相同的结构和节点值。 
//
// 例如: 
//给定的树 A: 
//
// 3 
// / \ 
// 4 5 
// / \ 
// 1 2 
//给定的树 B： 
//
// 4 
// / 
// 1 
//返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。 
//
// 示例 1： 
//
// 输入：A = [1,2,3], B = [3,1]
//输出：false
// 
//
// 示例 2： 
//
// 输入：A = [3,4,5,1,2], B = [4,1]
//输出：true 
//
// 限制： 
//
// 0 <= 节点个数 <= 10000 
// Related Topics 树 深度优先搜索 二叉树 
// 👍 321 👎 0


package leetcode.editor.cn1;

import com.leetcode.entity.TreeNode;

//Java：树的子结构

/**
 * 输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
 * <p>
 * B是A的子结构， 即 A中有出现和B相同的结构和节点值。
 * <p>
 * 例如:
 * 给定的树 A:
 * <p>
 *      3
 *     / \
 *    4   5
 *   / \
 *  1   2
 * 给定的树 B：
 * <p>
 *    4 
 *   /
 *  1
 * 返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。
 * <p>
 * 假设没有相同的节点
 */
public class ShuDeZiJieGouLcof_2 {
    public static void main(String[] args) {
        Solution solution = new ShuDeZiJieGouLcof_2().new Solution();
        // TO TEST
        TreeNode A = new TreeNode(3);
        A.setLeft(new TreeNode(4));
        A.setRight(new TreeNode(5));
        A.getLeft().setLeft(new TreeNode(1));
        A.getLeft().setRight(new TreeNode(2));

        TreeNode B = new TreeNode(4);
        B.setLeft(new TreeNode(1));
        System.out.println(solution.isSubStructure(A, B));
    }
    //leetcode submit region begin(Prohibit modification and deletion)

    /**
     * Definition for a binary tree node.
     * public class TreeNode {
     * int val;
     * TreeNode left;
     * TreeNode right;
     * TreeNode(int x) { val = x; }
     * }
     */
    class Solution {
        /**
         * 判断B是否A的子结构
         * 即B的根节点，在A中存在C，并且B的左子树 == C的左子树，B的右子树 == C的右子树
         *
         * @param A
         * @param B
         * @return
         */
        public boolean isSubStructure(TreeNode A, TreeNode B) {
            if (A == null || B == null)
                return false;
            return dfsNode(A, B);
        }

        public boolean dfsNode(TreeNode A, TreeNode B) {
            if (A != null && B != null) {
                boolean res = A.val == B.val;
                if (res) {
                    // 判断当前节点是否满足
                    res = isSameNode(A, B);
                }
                if (!res) {
                    // 不满足，查找左子树是否满足
                    res = dfsNode(A.left, B);
                }
                if (!res) {
                    // 不满足，查找右子树是否满足
                    res = dfsNode(A.right, B);
                }
                // 如果满足，直接返回true
                if (res)
                    return true;
            }
            return false;
        }

        /**
         * 判断两棵树是否相等
         *
         * @param node1
         * @param node2
         * @return
         */
        public boolean isSameNode(TreeNode node1, TreeNode node2) {
            if (node2 == null)
                return true;
            if (node1 == null)
                return false;
            if (node1.val != node2.val)
                return false;
            return isSameNode(node1.left, node2.left) && isSameNode(node1.right, node2.right);
        }
    }
    //leetcode submit region end(Prohibit modification and deletion)

}
